package demo2;

import org.junit.Test;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import java.nio.file.Paths;
import java.nio.file.StandardOpenOption;
import java.time.LocalDateTime;
import java.util.Date;
import java.util.Iterator;
import java.util.Set;

/**
 * 使用NIO完成网络通信的三个核心
 * <p>
 * 一：通道，主要负责连接
 * 二：缓冲区 负责数据的存取
 * 三：选择器 是SelectableChannel的多路复用器，用于监控SelectableChannel的一些状况
 */
public class TestNonBlockingIO {

    @Test
    public void client() throws IOException {
        /**
         * 获取通道
         */
        SocketChannel socketChannel = SocketChannel.open(new InetSocketAddress("127.0.0.1", 8080));
        /**
         * 切换成非阻塞模式
         */
        socketChannel.configureBlocking(false);
        /**
         * 配置指定大小的缓冲区
         */
        ByteBuffer buffer = ByteBuffer.allocate(1204);

        /**
         * 发送数据
         */
        buffer.put(new Date().toString().getBytes());
        buffer.flip();
        socketChannel.write(buffer);
        buffer.clear();
        socketChannel.close();

    }

    @Test
    public void server() throws IOException {
        /**
         * 获取通道
         */
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        /**
         * 切换非阻塞模式
         */
        serverSocketChannel.configureBlocking(false);

        /**
         * 绑定链接
         */
        serverSocketChannel.bind(new InetSocketAddress(8080));
        /**
         * 获取选择器
         */
        Selector selector = Selector.open();
        /**
         * 将通道注册到选择器上
         * 第二个参数是选择键  作用：表示选择器监视通道的什么状态（事件）
         */
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
        /**
         * 通过选择器轮询获取选择器上已经准备就绪的事件
         */
        while (selector.select() > 0) {
            /**
             * 获取当前选择器中所有注册的选择键（已注册-就绪的监听事件）
             */
            Set<SelectionKey> selectionKeys = selector.selectedKeys();
            Iterator<SelectionKey> selectionKeyIterator = selectionKeys.iterator();
            /**
             * 迭代获取准备就绪的事件
             */
            while (selectionKeyIterator.hasNext()) {
                SelectionKey selectionKey = selectionKeyIterator.next();
                /**
                 * 判断具体是什么事件准备就绪
                 */
                //接收就绪，得到客户端通道
                if (selectionKey.isAcceptable()) {
                    SocketChannel socketChannel = serverSocketChannel.accept();
                    /**
                     * 切换非阻塞模式
                     */
                    socketChannel.configureBlocking(false);
                    /**
                     * 将该通道注册到选择器上
                     */
                    socketChannel.register(selector, SelectionKey.OP_WRITE);

                    /**
                     * 如果是读就绪
                     */
                } else if (selectionKey.isReadable()) {
                    /**
                     * 获取当前选择器上读就绪状态的通道
                     */
                    SocketChannel schannel = (SocketChannel) selectionKey.channel();
                    int len = 0;
                    ByteBuffer byteBuffer = ByteBuffer.allocate(1024);
                    while ((len = schannel.read(byteBuffer)) > 0) {
                        byteBuffer.flip();
                        System.out.println(new String(byteBuffer.array(), 0, len));
                        byteBuffer.clear();

                    }
                }
                /**
                 * 取消选择键
                 */
                selectionKeyIterator.remove();
            }
        }

    }
}
